from sourcecode.dataset.dataset import FloorPlanDataset, to_color